-
Notifications
You must be signed in to change notification settings - Fork 314
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Publish precompiled wheels in Github release and add wheel index #365
Publish precompiled wheels in Github release and add wheel index #365
Conversation
I will try to finish addressing |
@liruilong940607 thank you very much for the feedback. I tried to address your comments. Any remaining concern or improvements suggestion before I publish the PR? |
wheels_per_local_version = {} | ||
for wheel in wheels: | ||
local_version = wheel["local_version"] | ||
if local_version not in wheels_per_local_version: |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
if local_version is not and and local_version not in wheels_per_local_version:
This looks great! Thanks for all the updates! I'm doing test now |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I tested it out and did some fixes here and there. Now I think it is good to go. Thanks a lot for digging into this @martinResearch !
The PR adds a mechanism to publish precompiled wheels in github releases and pypi and add a wheel index html file to be able to install precompiled wheels for a specific pytorch and cuda version using a single line like
One can also omit the
pt20cu118
and usebut then the full version needs to be specified to make sure one gets the right pytorch and cuda version.
The full list of pre-compiled wheel can be found here
Is that a solution you would potentially be happy to merge in your repository?
This PR also tries to upload all the wheels to pypi I am not sure pipy will allows us to upload wheels with "+" sign in the wheel name (i.e using local version to specify the cuda and pytorch versions), and I am not sure whe want to upload these wheels to pypi, so the upload step could be removed for the precompiled wheels